#from https://raw.githubusercontent.com/imohitmayank/got_network_viz_python/main/data/book1.csv
got_characters_book_1_string = """
Source,Target,Type,weight,book
Addam-Marbrand,Jaime-Lannister,Undirected,3,1
Addam-Marbrand,Tywin-Lannister,Undirected,6,1
Aegon-I-Targaryen,Daenerys-Targaryen,Undirected,5,1
Aegon-I-Targaryen,Eddard-Stark,Undirected,4,1
Aemon-Targaryen-(Maester-Aemon),Alliser-Thorne,Undirected,4,1
Aemon-Targaryen-(Maester-Aemon),Bowen-Marsh,Undirected,4,1
Aemon-Targaryen-(Maester-Aemon),Chett,Undirected,9,1
Aemon-Targaryen-(Maester-Aemon),Clydas,Undirected,5,1
Aemon-Targaryen-(Maester-Aemon),Jeor-Mormont,Undirected,13,1
Aemon-Targaryen-(Maester-Aemon),Jon-Snow,Undirected,34,1
Aemon-Targaryen-(Maester-Aemon),Samwell-Tarly,Undirected,5,1
Aerys-II-Targaryen,Brandon-Stark,Undirected,4,1
Aerys-II-Targaryen,Eddard-Stark,Undirected,10,1
Aerys-II-Targaryen,Gerold-Hightower,Undirected,3,1
Aerys-II-Targaryen,Jaime-Lannister,Undirected,5,1
Aerys-II-Targaryen,Jon-Arryn,Undirected,3,1
Aerys-II-Targaryen,Robert-Baratheon,Undirected,12,1
Aggo,Daenerys-Targaryen,Undirected,11,1
Aggo,Drogo,Undirected,6,1
Aggo,Jhogo,Undirected,4,1
Aggo,Jorah-Mormont,Undirected,5,1
Aggo,Quaro,Undirected,3,1
Aggo,Rakharo,Undirected,7,1
Albett,Halder,Undirected,3,1
Albett,Jon-Snow,Undirected,5,1
Albett,Rast,Undirected,3,1
Alliser-Thorne,Bowen-Marsh,Undirected,3,1
Alliser-Thorne,Grenn,Undirected,3,1
Alliser-Thorne,Halder,Undirected,4,1
Alliser-Thorne,Jeor-Mormont,Undirected,10,1
Alliser-Thorne,Jon-Snow,Undirected,32,1
Alliser-Thorne,Pypar,Undirected,4,1
Alliser-Thorne,Rast,Undirected,5,1
Alliser-Thorne,Samwell-Tarly,Undirected,8,1
Alliser-Thorne,Tyrion-Lannister,Undirected,7,1
Alyn,Eddard-Stark,Undirected,11,1
Alyn,Harwin,Undirected,6,1
Alyn,Jory-Cassel,Undirected,8,1
Alyn,Robert-Baratheon,Undirected,4,1
Alyn,Tomard,Undirected,3,1
Arthur-Dayne,Gerold-Hightower,Undirected,4,1
Arya-Stark,Benjen-Stark,Undirected,3,1
Arya-Stark,Bran-Stark,Undirected,14,1
Arya-Stark,Catelyn-Stark,Undirected,5,1
Arya-Stark,Cersei-Lannister,Undirected,12,1
Arya-Stark,Desmond,Undirected,3,1
Arya-Stark,Eddard-Stark,Undirected,30,1
Arya-Stark,Ilyn-Payne,Undirected,5,1
Arya-Stark,Jeyne-Poole,Undirected,10,1
Arya-Stark,Joffrey-Baratheon,Undirected,39,1
Arya-Stark,Jon-Snow,Undirected,37,1
Arya-Stark,Jory-Cassel,Undirected,12,1
Arya-Stark,Meryn-Trant,Undirected,4,1
Arya-Stark,Mordane,Undirected,39,1
Arya-Stark,Mycah,Undirected,18,1
Arya-Stark,Myrcella-Baratheon,Undirected,13,1
Arya-Stark,Petyr-Baelish,Undirected,3,1
Arya-Stark,Rickon-Stark,Undirected,7,1
Arya-Stark,Robb-Stark,Undirected,15,1
Arya-Stark,Robert-Baratheon,Undirected,8,1
Arya-Stark,Rodrik-Cassel,Undirected,3,1
Arya-Stark,Sandor-Clegane,Undirected,7,1
Arya-Stark,Sansa-Stark,Undirected,104,1
Arya-Stark,Syrio-Forel,Undirected,20,1
Arya-Stark,Tomard,Undirected,6,1
Arya-Stark,Tommen-Baratheon,Undirected,5,1
Arya-Stark,Vayon-Poole,Undirected,4,1
Arya-Stark,Yoren,Undirected,4,1
Arys-Oakheart,Cersei-Lannister,Undirected,3,1
Balon-Greyjoy,Eddard-Stark,Undirected,3,1
Balon-Greyjoy,Robert-Baratheon,Undirected,4,1
Balon-Swann,Renly-Baratheon,Undirected,3,1
Balon-Swann,Sandor-Clegane,Undirected,3,1
Barristan-Selmy,Boros-Blount,Undirected,3,1
Barristan-Selmy,Cersei-Lannister,Undirected,12,1
Barristan-Selmy,Eddard-Stark,Undirected,28,1
Barristan-Selmy,Jaime-Lannister,Undirected,4,1
Barristan-Selmy,Joffrey-Baratheon,Undirected,6,1
Barristan-Selmy,Jon-Arryn,Undirected,4,1
Barristan-Selmy,Meryn-Trant,Undirected,3,1
Barristan-Selmy,Petyr-Baelish,Undirected,7,1
Barristan-Selmy,Pycelle,Undirected,8,1
Barristan-Selmy,Renly-Baratheon,Undirected,13,1
Barristan-Selmy,Robert-Baratheon,Undirected,40,1
Barristan-Selmy,Sandor-Clegane,Undirected,3,1
Barristan-Selmy,Sansa-Stark,Undirected,3,1
Barristan-Selmy,Varys,Undirected,9,1
Benjen-Stark,Bran-Stark,Undirected,6,1
Benjen-Stark,Cersei-Lannister,Undirected,3,1
Benjen-Stark,Jaremy-Rykker,Undirected,3,1
Benjen-Stark,Jeor-Mormont,Undirected,9,1
Benjen-Stark,Jon-Snow,Undirected,41,1
Benjen-Stark,Luwin,Undirected,7,1
Benjen-Stark,Mance-Rayder,Undirected,3,1
Benjen-Stark,Robb-Stark,Undirected,7,1
Benjen-Stark,Robert-Baratheon,Undirected,3,1
Benjen-Stark,Theon-Greyjoy,Undirected,4,1
Benjen-Stark,Tyrion-Lannister,Undirected,8,1
Benjen-Stark,Waymar-Royce,Undirected,3,1
Benjen-Stark,Yoren,Undirected,4,1
Beric-Dondarrion,Eddard-Stark,Undirected,3,1
Beric-Dondarrion,Gregor-Clegane,Undirected,6,1
Beric-Dondarrion,Loras-Tyrell,Undirected,5,1
Beric-Dondarrion,Robert-Baratheon,Undirected,5,1
Beric-Dondarrion,Sansa-Stark,Undirected,4,1
Beric-Dondarrion,Thoros-of-Myr,Undirected,5,1
Boros-Blount,Cersei-Lannister,Undirected,9,1
Boros-Blount,Joffrey-Baratheon,Undirected,3,1
Boros-Blount,Meryn-Trant,Undirected,5,1
Boros-Blount,Sansa-Stark,Undirected,6,1
Bowen-Marsh,Jeor-Mormont,Undirected,8,1
Bowen-Marsh,Jon-Snow,Undirected,10,1
Bowen-Marsh,Samwell-Tarly,Undirected,3,1
Bran-Stark,Catelyn-Stark,Undirected,18,1
Bran-Stark,Cersei-Lannister,Undirected,6,1
Bran-Stark,Eddard-Stark,Undirected,19,1
Bran-Stark,Hali,Undirected,3,1
Bran-Stark,Hallis-Mollen,Undirected,3,1
Bran-Stark,Hodor,Undirected,29,1
Bran-Stark,Hullen,Undirected,3,1
Bran-Stark,Jeor-Mormont,Undirected,4,1
Bran-Stark,Joffrey-Baratheon,Undirected,3,1
Bran-Stark,Jon-Arryn,Undirected,7,1
Bran-Stark,Jon-Snow,Undirected,56,1
Bran-Stark,Jory-Cassel,Undirected,11,1
Bran-Stark,Joseth,Undirected,4,1
Bran-Stark,Luwin,Undirected,65,1
Bran-Stark,Nan,Undirected,17,1
Bran-Stark,Osha,Undirected,12,1
Bran-Stark,Petyr-Baelish,Undirected,3,1
Bran-Stark,Renly-Baratheon,Undirected,3,1
Bran-Stark,Rickard-Karstark,Undirected,4,1
Bran-Stark,Rickard-Stark,Undirected,4,1
Bran-Stark,Rickon-Stark,Undirected,32,1
Bran-Stark,Robb-Stark,Undirected,112,1
Bran-Stark,Robert-Baratheon,Undirected,11,1
Bran-Stark,Rodrik-Cassel,Undirected,6,1
Bran-Stark,Sansa-Stark,Undirected,10,1
Bran-Stark,Stiv,Undirected,4,1
Bran-Stark,Theon-Greyjoy,Undirected,23,1
Bran-Stark,Tommen-Baratheon,Undirected,5,1
Bran-Stark,Tyrion-Lannister,Undirected,31,1
Bran-Stark,Yoren,Undirected,3,1
Brandon-Stark,Eddard-Stark,Undirected,11,1
Brandon-Stark,Lyanna-Stark,Undirected,4,1
Brandon-Stark,Petyr-Baelish,Undirected,14,1
Brandon-Stark,Rickard-Stark,Undirected,3,1
Brandon-Stark,Robert-Baratheon,Undirected,5,1
Bronn,Catelyn-Stark,Undirected,6,1
Bronn,Chiggen,Undirected,8,1
Bronn,Marillion,Undirected,4,1
Bronn,Rodrik-Cassel,Undirected,4,1
Bronn,Shae,Undirected,3,1
Bronn,Shagga,Undirected,3,1
Bronn,Tyrion-Lannister,Undirected,61,1
Bronn,Tywin-Lannister,Undirected,4,1
Bronn,Vardis-Egen,Undirected,11,1
Bronn,Willis-Wode,Undirected,5,1
Brynden-Tully,Catelyn-Stark,Undirected,16,1
Brynden-Tully,Edmure-Tully,Undirected,3,1
Brynden-Tully,Hoster-Tully,Undirected,5,1
Brynden-Tully,Jaime-Lannister,Undirected,7,1
Brynden-Tully,Lysa-Arryn,Undirected,10,1
Brynden-Tully,Nestor-Royce,Undirected,3,1
Brynden-Tully,Robb-Stark,Undirected,9,1
Brynden-Tully,Theon-Greyjoy,Undirected,6,1
Brynden-Tully,Tyrion-Lannister,Undirected,4,1
Brynden-Tully,Tywin-Lannister,Undirected,3,1
Brynden-Tully,Walder-Frey,Undirected,3,1
Catelyn-Stark,Cersei-Lannister,Undirected,12,1
Catelyn-Stark,Colemon,Undirected,3,1
Catelyn-Stark,Donnel-Waynwood,Undirected,3,1
Catelyn-Stark,Eddard-Stark,Undirected,64,1
Catelyn-Stark,Edmure-Tully,Undirected,7,1
Catelyn-Stark,Eon-Hunter,Undirected,3,1
Catelyn-Stark,Hallis-Mollen,Undirected,10,1
Catelyn-Stark,Hoster-Tully,Undirected,6,1
Catelyn-Stark,Jaime-Lannister,Undirected,9,1
Catelyn-Stark,Joffrey-Baratheon,Undirected,4,1
Catelyn-Stark,Jon-Arryn,Undirected,11,1
Catelyn-Stark,Jon-Snow,Undirected,14,1
Catelyn-Stark,Jon-Umber-(Greatjon),Undirected,4,1
Catelyn-Stark,Luwin,Undirected,14,1
Catelyn-Stark,Lysa-Arryn,Undirected,33,1
Catelyn-Stark,Marillion,Undirected,6,1
Catelyn-Stark,Masha-Heddle,Undirected,5,1
Catelyn-Stark,Moreo-Tumitis,Undirected,3,1
Catelyn-Stark,Mya-Stone,Undirected,9,1
Catelyn-Stark,Mychel-Redfort,Undirected,3,1
Catelyn-Stark,Nestor-Royce,Undirected,3,1
Catelyn-Stark,Petyr-Baelish,Undirected,22,1
Catelyn-Stark,Rickard-Karstark,Undirected,6,1
Catelyn-Stark,Rickon-Stark,Undirected,3,1
Catelyn-Stark,Robb-Stark,Undirected,49,1
Catelyn-Stark,Robert-Arryn,Undirected,3,1
Catelyn-Stark,Robert-Baratheon,Undirected,22,1
Catelyn-Stark,Rodrik-Cassel,Undirected,25,1
Catelyn-Stark,Sansa-Stark,Undirected,9,1
Catelyn-Stark,Stevron-Frey,Undirected,5,1
Catelyn-Stark,Theon-Greyjoy,Undirected,9,1
Catelyn-Stark,Tyrion-Lannister,Undirected,49,1
Catelyn-Stark,Tytos-Blackwood,Undirected,4,1
Catelyn-Stark,Tywin-Lannister,Undirected,7,1
Catelyn-Stark,Vardis-Egen,Undirected,5,1
Catelyn-Stark,Varys,Undirected,10,1
Catelyn-Stark,Walder-Frey,Undirected,13,1
Catelyn-Stark,Wendel-Manderly,Undirected,3,1
Catelyn-Stark,Willis-Wode,Undirected,5,1
Cayn,Eddard-Stark,Undirected,7,1
Cayn,Renly-Baratheon,Undirected,3,1
Cayn,Tomard,Undirected,6,1
Cersei-Lannister,Eddard-Stark,Undirected,69,1
Cersei-Lannister,Ilyn-Payne,Undirected,4,1
Cersei-Lannister,Jaime-Lannister,Undirected,16,1
Cersei-Lannister,Janos-Slynt,Undirected,4,1
Cersei-Lannister,Joffrey-Baratheon,Undirected,42,1
Cersei-Lannister,Jon-Arryn,Undirected,7,1
Cersei-Lannister,Jon-Snow,Undirected,4,1
Cersei-Lannister,Jory-Cassel,Undirected,5,1
Cersei-Lannister,Meryn-Trant,Undirected,4,1
Cersei-Lannister,Mordane,Undirected,3,1
Cersei-Lannister,Myrcella-Baratheon,Undirected,7,1
Cersei-Lannister,Petyr-Baelish,Undirected,13,1
Cersei-Lannister,Pycelle,Undirected,9,1
Cersei-Lannister,Renly-Baratheon,Undirected,5,1
Cersei-Lannister,Robert-Baratheon,Undirected,72,1
Cersei-Lannister,Sandor-Clegane,Undirected,5,1
Cersei-Lannister,Sansa-Stark,Undirected,29,1
Cersei-Lannister,Stannis-Baratheon,Undirected,5,1
Cersei-Lannister,Tommen-Baratheon,Undirected,4,1
Cersei-Lannister,Tyrion-Lannister,Undirected,27,1
Cersei-Lannister,Tywin-Lannister,Undirected,10,1
Cersei-Lannister,Varys,Undirected,19,1
Cersei-Lannister,Vayon-Poole,Undirected,4,1
Chella,Tyrion-Lannister,Undirected,5,1
Chett,Jon-Snow,Undirected,4,1
Chett,Samwell-Tarly,Undirected,6,1
Chiggen,Tyrion-Lannister,Undirected,5,1
Chiggen,Willis-Wode,Undirected,3,1
Clement-Piper,Karyl-Vance,Undirected,4,1
Cohollo,Daenerys-Targaryen,Undirected,7,1
Cohollo,Drogo,Undirected,9,1
Cohollo,Haggo,Undirected,7,1
Cohollo,Qotho,Undirected,6,1
Colemon,Vardis-Egen,Undirected,4,1
Conn,Coratt,Undirected,5,1
Conn,Shagga,Undirected,9,1
Conn,Tyrion-Lannister,Undirected,6,1
Daenerys-Targaryen,Doreah,Undirected,15,1
Daenerys-Targaryen,Drogo,Undirected,101,1
Daenerys-Targaryen,Eddard-Stark,Undirected,5,1
Daenerys-Targaryen,Eroeh,Undirected,5,1
Daenerys-Targaryen,Haggo,Undirected,12,1
Daenerys-Targaryen,Illyrio-Mopatis,Undirected,24,1
Daenerys-Targaryen,Irri,Undirected,25,1
Daenerys-Targaryen,Jhiqui,Undirected,26,1
Daenerys-Targaryen,Jhogo,Undirected,17,1
Daenerys-Targaryen,Jorah-Mormont,Undirected,75,1
Daenerys-Targaryen,Mirri-Maz-Duur,Undirected,31,1
Daenerys-Targaryen,Qotho,Undirected,18,1
Daenerys-Targaryen,Quaro,Undirected,5,1
Daenerys-Targaryen,Rakharo,Undirected,4,1
Daenerys-Targaryen,Rhaegar-Targaryen,Undirected,4,1
Daenerys-Targaryen,Robert-Baratheon,Undirected,11,1
Daenerys-Targaryen,Varys,Undirected,5,1
Daenerys-Targaryen,Viserys-Targaryen,Undirected,37,1
Danwell-Frey,Hosteen-Frey,Undirected,3,1
Danwell-Frey,Jared-Frey,Undirected,3,1
Danwell-Frey,Walder-Frey,Undirected,3,1
Dareon,Jon-Snow,Undirected,6,1
Daryn-Hornwood,Eddard-Stark,Undirected,3,1
Daryn-Hornwood,Torrhen-Karstark,Undirected,3,1
Desmond,Eddard-Stark,Undirected,4,1
Dolf,Shagga,Undirected,7,1
Donal-Noye,Jon-Snow,Undirected,7,1
Doreah,Drogo,Undirected,5,1
Doreah,Irri,Undirected,14,1
Doreah,Jhiqui,Undirected,11,1
Doreah,Jorah-Mormont,Undirected,3,1
Doreah,Viserys-Targaryen,Undirected,3,1
Drogo,Haggo,Undirected,9,1
Drogo,Illyrio-Mopatis,Undirected,16,1
Drogo,Irri,Undirected,9,1
Drogo,Jhiqui,Undirected,5,1
Drogo,Jhogo,Undirected,7,1
Drogo,Jommo,Undirected,3,1
Drogo,Jorah-Mormont,Undirected,16,1
Drogo,Mirri-Maz-Duur,Undirected,12,1
Drogo,Ogo,Undirected,11,1
Drogo,Qotho,Undirected,13,1
Drogo,Quaro,Undirected,3,1
Drogo,Rakharo,Undirected,4,1
Drogo,Rhaego,Undirected,5,1
Drogo,Robert-Baratheon,Undirected,3,1
Drogo,Viserys-Targaryen,Undirected,19,1
Dywen,Jon-Snow,Undirected,4,1
Eddard-Stark,Edmure-Tully,Undirected,3,1
Eddard-Stark,Galbart-Glover,Undirected,4,1
Eddard-Stark,Gendry,Undirected,3,1
Eddard-Stark,Gerold-Hightower,Undirected,3,1
Eddard-Stark,Gregor-Clegane,Undirected,10,1
Eddard-Stark,Harwin,Undirected,5,1
Eddard-Stark,High-Septon-(fat_one),Undirected,4,1
Eddard-Stark,Howland-Reed,Undirected,3,1
Eddard-Stark,Ilyn-Payne,Undirected,6,1
Eddard-Stark,Jacks,Undirected,4,1
Eddard-Stark,Jaime-Lannister,Undirected,27,1
Eddard-Stark,Janos-Slynt,Undirected,4,1
Eddard-Stark,Joffrey-Baratheon,Undirected,26,1
Eddard-Stark,Jon-Arryn,Undirected,47,1
Eddard-Stark,Jon-Snow,Undirected,38,1
Eddard-Stark,Jory-Cassel,Undirected,32,1
Eddard-Stark,Joss,Undirected,5,1
Eddard-Stark,Loras-Tyrell,Undirected,12,1
Eddard-Stark,Luwin,Undirected,16,1
Eddard-Stark,Lyanna-Stark,Undirected,19,1
Eddard-Stark,Lysa-Arryn,Undirected,12,1
Eddard-Stark,Marq-Piper,Undirected,3,1
Eddard-Stark,Mordane,Undirected,11,1
Eddard-Stark,Mycah,Undirected,3,1
Eddard-Stark,Petyr-Baelish,Undirected,81,1
Eddard-Stark,Porther,Undirected,4,1
Eddard-Stark,Pycelle,Undirected,33,1
Eddard-Stark,Raymun-Darry,Undirected,4,1
Eddard-Stark,Renly-Baratheon,Undirected,33,1
Eddard-Stark,Rhaegar-Targaryen,Undirected,10,1
Eddard-Stark,Rickard-Karstark,Undirected,4,1
Eddard-Stark,Rickon-Stark,Undirected,4,1
Eddard-Stark,Robb-Stark,Undirected,16,1
Eddard-Stark,Robert-Baratheon,Undirected,291,1
Eddard-Stark,Rodrik-Cassel,Undirected,7,1
Eddard-Stark,Sandor-Clegane,Undirected,10,1
Eddard-Stark,Sansa-Stark,Undirected,34,1
Eddard-Stark,Stannis-Baratheon,Undirected,18,1
Eddard-Stark,Syrio-Forel,Undirected,6,1
Eddard-Stark,Theon-Greyjoy,Undirected,5,1
Eddard-Stark,Tobho-Mott,Undirected,7,1
Eddard-Stark,Tomard,Undirected,15,1
Eddard-Stark,Torrhen-Karstark,Undirected,5,1
Eddard-Stark,Tregar,Undirected,4,1
Eddard-Stark,Tyrion-Lannister,Undirected,21,1
Eddard-Stark,Tywin-Lannister,Undirected,18,1
Eddard-Stark,Varly,Undirected,4,1
Eddard-Stark,Varys,Undirected,61,1
Eddard-Stark,Vayon-Poole,Undirected,12,1
Eddard-Stark,Wyl-(guard),Undirected,3,1
Eddard-Stark,Wylla,Undirected,3,1
Edmure-Tully,Hoster-Tully,Undirected,7,1
Edmure-Tully,Jaime-Lannister,Undirected,4,1
Edmure-Tully,Karyl-Vance,Undirected,4,1
Edmure-Tully,Lysa-Arryn,Undirected,5,1
Edmure-Tully,Marq-Piper,Undirected,3,1
Edmure-Tully,Petyr-Baelish,Undirected,4,1
Edmure-Tully,Robb-Stark,Undirected,5,1
Edmure-Tully,Tytos-Blackwood,Undirected,3,1
Eon-Hunter,Lysa-Arryn,Undirected,3,1
Eon-Hunter,Vardis-Egen,Undirected,3,1
Fogo,Ogo,Undirected,4,1
Galbart-Glover,Jaime-Lannister,Undirected,3,1
Galbart-Glover,Jon-Umber-(Greatjon),Undirected,4,1
Galbart-Glover,Rickard-Karstark,Undirected,4,1
Galbart-Glover,Robb-Stark,Undirected,5,1
Galbart-Glover,Roose-Bolton,Undirected,3,1
Gared,Jeor-Mormont,Undirected,4,1
Gared,Waymar-Royce,Undirected,13,1
Gared,Will-(prologue),Undirected,12,1
Gerold-Hightower,Oswell-Whent,Undirected,3,1
Gerold-Hightower,Robert-Baratheon,Undirected,4,1
Gregor-Clegane,Jaime-Lannister,Undirected,3,1
Gregor-Clegane,Loras-Tyrell,Undirected,10,1
Gregor-Clegane,Marq-Piper,Undirected,6,1
Gregor-Clegane,Petyr-Baelish,Undirected,3,1
Gregor-Clegane,Pycelle,Undirected,3,1
Gregor-Clegane,Robert-Baratheon,Undirected,4,1
Gregor-Clegane,Sandor-Clegane,Undirected,11,1
Gregor-Clegane,Sansa-Stark,Undirected,5,1
Gregor-Clegane,Tyrion-Lannister,Undirected,7,1
Gregor-Clegane,Tywin-Lannister,Undirected,6,1
Grenn,Halder,Undirected,6,1
Grenn,Jeor-Mormont,Undirected,3,1
Grenn,Jon-Snow,Undirected,31,1
Grenn,Pypar,Undirected,27,1
Grenn,Samwell-Tarly,Undirected,8,1
Grenn,Todder,Undirected,7,1
Gunthor-son-of-Gurn,Tyrion-Lannister,Undirected,7,1
Haggo,Mirri-Maz-Duur,Undirected,3,1
Haggo,Qotho,Undirected,11,1
Haggo,Rakharo,Undirected,3,1
Halder,Jon-Snow,Undirected,23,1
Halder,Pypar,Undirected,8,1
Halder,Rast,Undirected,3,1
Halder,Samwell-Tarly,Undirected,6,1
Halder,Todder,Undirected,3,1
Hallis-Mollen,Luwin,Undirected,4,1
Hallis-Mollen,Robb-Stark,Undirected,13,1
Hallis-Mollen,Theon-Greyjoy,Undirected,7,1
Harwin,Hullen,Undirected,3,1
Harwin,Jory-Cassel,Undirected,4,1
Harys-Swyft,Jaime-Lannister,Undirected,3,1
Heward,Jory-Cassel,Undirected,5,1
Heward,Wyl-(guard),Undirected,3,1
High-Septon-(fat_one),Joffrey-Baratheon,Undirected,4,1
Hobb,Jon-Snow,Undirected,5,1
Hodor,Luwin,Undirected,10,1
Hodor,Nan,Undirected,8,1
Hodor,Robb-Stark,Undirected,5,1
Hosteen-Frey,Jared-Frey,Undirected,3,1
Hoster-Tully,Lysa-Arryn,Undirected,3,1
Hoster-Tully,Robert-Baratheon,Undirected,3,1
Hugh,Jon-Arryn,Undirected,3,1
Hullen,Jory-Cassel,Undirected,5,1
Hullen,Luwin,Undirected,3,1
Hullen,Robb-Stark,Undirected,3,1
Hullen,Rodrik-Cassel,Undirected,3,1
Illyrio-Mopatis,Irri,Undirected,3,1
Illyrio-Mopatis,Jorah-Mormont,Undirected,7,1
Illyrio-Mopatis,Robert-Baratheon,Undirected,6,1
Illyrio-Mopatis,Viserys-Targaryen,Undirected,14,1
Ilyn-Payne,Jeyne-Poole,Undirected,3,1
Ilyn-Payne,Joffrey-Baratheon,Undirected,4,1
Ilyn-Payne,Loras-Tyrell,Undirected,3,1
Ilyn-Payne,Robert-Baratheon,Undirected,4,1
Ilyn-Payne,Sansa-Stark,Undirected,9,1
Ilyn-Payne,Varys,Undirected,5,1
Irri,Jhiqui,Undirected,14,1
Irri,Jhogo,Undirected,7,1
Irri,Jorah-Mormont,Undirected,4,1
Irri,Viserys-Targaryen,Undirected,5,1
Jafer-Flowers,Jaremy-Rykker,Undirected,3,1
Jaime-Lannister,Joffrey-Baratheon,Undirected,12,1
Jaime-Lannister,Jon-Arryn,Undirected,5,1
Jaime-Lannister,Jory-Cassel,Undirected,4,1
Jaime-Lannister,Kevan-Lannister,Undirected,4,1
Jaime-Lannister,Loras-Tyrell,Undirected,3,1
Jaime-Lannister,Lysa-Arryn,Undirected,6,1
Jaime-Lannister,Petyr-Baelish,Undirected,7,1
Jaime-Lannister,Renly-Baratheon,Undirected,4,1
Jaime-Lannister,Rickard-Karstark,Undirected,3,1
Jaime-Lannister,Robb-Stark,Undirected,9,1
Jaime-Lannister,Robert-Baratheon,Undirected,24,1
Jaime-Lannister,Rodrik-Cassel,Undirected,3,1
Jaime-Lannister,Sandor-Clegane,Undirected,12,1
Jaime-Lannister,Sansa-Stark,Undirected,3,1
Jaime-Lannister,Tyrion-Lannister,Undirected,36,1
Jaime-Lannister,Tywin-Lannister,Undirected,16,1
Jaime-Lannister,Varys,Undirected,3,1
Jaime-Lannister,Wyl-(guard),Undirected,3,1
Janos-Slynt,Joffrey-Baratheon,Undirected,4,1
Janos-Slynt,Petyr-Baelish,Undirected,3,1
Janos-Slynt,Pycelle,Undirected,3,1
Janos-Slynt,Robert-Baratheon,Undirected,4,1
Janos-Slynt,Sansa-Stark,Undirected,3,1
Jaremy-Rykker,Jeor-Mormont,Undirected,11,1
Jaremy-Rykker,Jon-Snow,Undirected,7,1
Jeor-Mormont,Jon-Snow,Undirected,81,1
Jeor-Mormont,Robert-Baratheon,Undirected,4,1
Jeor-Mormont,Samwell-Tarly,Undirected,8,1
Jeor-Mormont,Tyrion-Lannister,Undirected,23,1
Jeor-Mormont,Yoren,Undirected,3,1
Jeyne-Poole,Loras-Tyrell,Undirected,4,1
Jeyne-Poole,Mordane,Undirected,7,1
Jeyne-Poole,Robert-Baratheon,Undirected,3,1
Jeyne-Poole,Sansa-Stark,Undirected,27,1
Jhiqui,Jorah-Mormont,Undirected,4,1
Jhogo,Jorah-Mormont,Undirected,5,1
Jhogo,Mirri-Maz-Duur,Undirected,3,1
Joffrey-Baratheon,Jon-Snow,Undirected,10,1
Joffrey-Baratheon,Loras-Tyrell,Undirected,3,1
Joffrey-Baratheon,Meryn-Trant,Undirected,3,1
Joffrey-Baratheon,Mordane,Undirected,9,1
Joffrey-Baratheon,Mycah,Undirected,14,1
Joffrey-Baratheon,Myrcella-Baratheon,Undirected,5,1
Joffrey-Baratheon,Petyr-Baelish,Undirected,11,1
Joffrey-Baratheon,Pycelle,Undirected,4,1
Joffrey-Baratheon,Renly-Baratheon,Undirected,9,1
Joffrey-Baratheon,Robb-Stark,Undirected,19,1
Joffrey-Baratheon,Robert-Baratheon,Undirected,30,1
Joffrey-Baratheon,Rodrik-Cassel,Undirected,9,1
Joffrey-Baratheon,Sandor-Clegane,Undirected,24,1
Joffrey-Baratheon,Sansa-Stark,Undirected,87,1
Joffrey-Baratheon,Stannis-Baratheon,Undirected,7,1
Joffrey-Baratheon,Tommen-Baratheon,Undirected,8,1
Joffrey-Baratheon,Tyrion-Lannister,Undirected,13,1
Joffrey-Baratheon,Tywin-Lannister,Undirected,3,1
Joffrey-Baratheon,Varys,Undirected,7,1
Jommo,Ogo,Undirected,3,1
Jon-Arryn,Lysa-Arryn,Undirected,12,1
Jon-Arryn,Petyr-Baelish,Undirected,4,1
Jon-Arryn,Pycelle,Undirected,11,1
Jon-Arryn,Robert-Baratheon,Undirected,32,1
Jon-Arryn,Stannis-Baratheon,Undirected,13,1
Jon-Arryn,Tyrion-Lannister,Undirected,5,1
Jon-Arryn,Varys,Undirected,5,1
Jon-Snow,Jory-Cassel,Undirected,3,1
Jon-Snow,Luwin,Undirected,10,1
Jon-Snow,Matthar,Undirected,3,1
Jon-Snow,Mordane,Undirected,4,1
Jon-Snow,Othor,Undirected,5,1
Jon-Snow,Pypar,Undirected,45,1
Jon-Snow,Rast,Undirected,8,1
Jon-Snow,Rickon-Stark,Undirected,11,1
Jon-Snow,Robb-Stark,Undirected,53,1
Jon-Snow,Robert-Baratheon,Undirected,18,1
Jon-Snow,Rodrik-Cassel,Undirected,3,1
Jon-Snow,Samwell-Tarly,Undirected,81,1
Jon-Snow,Sansa-Stark,Undirected,8,1
Jon-Snow,Theon-Greyjoy,Undirected,8,1
Jon-Snow,Todder,Undirected,13,1
Jon-Snow,Tyrion-Lannister,Undirected,56,1
Jon-Snow,Yoren,Undirected,6,1
Jon-Umber-(Greatjon),Maege-Mormont,Undirected,5,1
Jon-Umber-(Greatjon),Rickard-Karstark,Undirected,7,1
Jon-Umber-(Greatjon),Robb-Stark,Undirected,8,1
Jon-Umber-(Greatjon),Roose-Bolton,Undirected,3,1
Jon-Umber-(Greatjon),Theon-Greyjoy,Undirected,5,1
Jon-Umber-(Greatjon),Tywin-Lannister,Undirected,3,1
Jonos-Bracken,Tytos-Blackwood,Undirected,3,1
Jorah-Mormont,Mirri-Maz-Duur,Undirected,5,1
Jorah-Mormont,Qotho,Undirected,7,1
Jorah-Mormont,Rhaegar-Targaryen,Undirected,5,1
Jorah-Mormont,Robert-Baratheon,Undirected,3,1
Jorah-Mormont,Viserys-Targaryen,Undirected,15,1
Jory-Cassel,Mordane,Undirected,3,1
Jory-Cassel,Petyr-Baelish,Undirected,8,1
Jory-Cassel,Robb-Stark,Undirected,9,1
Jory-Cassel,Robert-Baratheon,Undirected,7,1
Jory-Cassel,Rodrik-Cassel,Undirected,5,1
Jory-Cassel,Sansa-Stark,Undirected,9,1
Jory-Cassel,Stannis-Baratheon,Undirected,3,1
Jory-Cassel,Theon-Greyjoy,Undirected,3,1
Jory-Cassel,Tomard,Undirected,5,1
Jory-Cassel,Vayon-Poole,Undirected,5,1
Jory-Cassel,Wyl-(guard),Undirected,6,1
Joseth,Luwin,Undirected,3,1
Jyck,Morrec,Undirected,5,1
Jyck,Tyrion-Lannister,Undirected,6,1
Karyl-Vance,Marq-Piper,Undirected,3,1
Kevan-Lannister,Tyrion-Lannister,Undirected,15,1
Kevan-Lannister,Tywin-Lannister,Undirected,9,1
Kurleket,Tyrion-Lannister,Undirected,5,1
Lancel-Lannister,Robert-Baratheon,Undirected,3,1
Leo-Lefford,Tyrion-Lannister,Undirected,4,1
Loras-Tyrell,Mace-Tyrell,Undirected,3,1
Loras-Tyrell,Petyr-Baelish,Undirected,5,1
Loras-Tyrell,Renly-Baratheon,Undirected,5,1
Loras-Tyrell,Robert-Baratheon,Undirected,4,1
Loras-Tyrell,Rodrik-Cassel,Undirected,3,1
Loras-Tyrell,Sandor-Clegane,Undirected,4,1
Loras-Tyrell,Sansa-Stark,Undirected,14,1
Luwin,Nan,Undirected,4,1
Luwin,Osha,Undirected,16,1
Luwin,Rickard-Karstark,Undirected,3,1
Luwin,Rickon-Stark,Undirected,9,1
Luwin,Robb-Stark,Undirected,20,1
Luwin,Rodrik-Cassel,Undirected,3,1
Luwin,Theon-Greyjoy,Undirected,10,1
Luwin,Tommen-Baratheon,Undirected,3,1
Luwin,Tyrion-Lannister,Undirected,10,1
Lyanna-Stark,Robert-Baratheon,Undirected,9,1
Lyanna-Stark,Sansa-Stark,Undirected,4,1
Lyn-Corbray,Lysa-Arryn,Undirected,3,1
Lysa-Arryn,Petyr-Baelish,Undirected,4,1
Lysa-Arryn,Robert-Arryn,Undirected,8,1
Lysa-Arryn,Robert-Baratheon,Undirected,5,1
Lysa-Arryn,Stannis-Baratheon,Undirected,4,1
Lysa-Arryn,Tyrion-Lannister,Undirected,25,1
Lysa-Arryn,Vardis-Egen,Undirected,5,1
Mace-Tyrell,Paxter-Redwyne,Undirected,3,1
Maege-Mormont,Rickard-Karstark,Undirected,3,1
Maegor-I-Targaryen,Sansa-Stark,Undirected,4,1
Marillion,Rodrik-Cassel,Undirected,5,1
Marillion,Tyrion-Lannister,Undirected,10,1
Marillion,Willis-Wode,Undirected,4,1
Marq-Piper,Tywin-Lannister,Undirected,3,1
Meryn-Trant,Robert-Baratheon,Undirected,3,1
Meryn-Trant,Sandor-Clegane,Undirected,4,1
Meryn-Trant,Sansa-Stark,Undirected,5,1
Meryn-Trant,Syrio-Forel,Undirected,5,1
Mirri-Maz-Duur,Qotho,Undirected,9,1
Mord,Tyrion-Lannister,Undirected,20,1
Mordane,Myrcella-Baratheon,Undirected,12,1
Mordane,Petyr-Baelish,Undirected,3,1
Mordane,Robert-Baratheon,Undirected,7,1
Mordane,Sansa-Stark,Undirected,43,1
Morrec,Tyrion-Lannister,Undirected,5,1
Mycah,Sandor-Clegane,Undirected,5,1
Mycah,Sansa-Stark,Undirected,10,1
Myrcella-Baratheon,Robert-Baratheon,Undirected,3,1
Myrcella-Baratheon,Sansa-Stark,Undirected,8,1
Myrcella-Baratheon,Tommen-Baratheon,Undirected,6,1
Nan,Robb-Stark,Undirected,4,1
Osha,Robb-Stark,Undirected,3,1
Petyr-Baelish,Pycelle,Undirected,14,1
Petyr-Baelish,Renly-Baratheon,Undirected,12,1
Petyr-Baelish,Robert-Baratheon,Undirected,30,1
Petyr-Baelish,Rodrik-Cassel,Undirected,7,1
Petyr-Baelish,Sandor-Clegane,Undirected,3,1
Petyr-Baelish,Sansa-Stark,Undirected,8,1
Petyr-Baelish,Tomard,Undirected,4,1
Petyr-Baelish,Tyrion-Lannister,Undirected,11,1
Petyr-Baelish,Varys,Undirected,29,1
Pycelle,Renly-Baratheon,Undirected,7,1
Pycelle,Robert-Baratheon,Undirected,26,1
Pycelle,Sansa-Stark,Undirected,8,1
Pycelle,Stannis-Baratheon,Undirected,3,1
Pycelle,Tywin-Lannister,Undirected,4,1
Pycelle,Varys,Undirected,17,1
Pypar,Rast,Undirected,4,1
Pypar,Samwell-Tarly,Undirected,11,1
Pypar,Todder,Undirected,8,1
Qotho,Quaro,Undirected,4,1
Quaro,Rakharo,Undirected,6,1
Randyll-Tarly,Samwell-Tarly,Undirected,5,1
Rast,Samwell-Tarly,Undirected,3,1
Raymun-Darry,Robert-Baratheon,Undirected,8,1
Renly-Baratheon,Robb-Stark,Undirected,3,1
Renly-Baratheon,Robert-Baratheon,Undirected,42,1
Renly-Baratheon,Sandor-Clegane,Undirected,7,1
Renly-Baratheon,Sansa-Stark,Undirected,5,1
Renly-Baratheon,Stannis-Baratheon,Undirected,19,1
Renly-Baratheon,Tyrion-Lannister,Undirected,4,1
Renly-Baratheon,Varys,Undirected,9,1
Rhaegar-Targaryen,Robert-Baratheon,Undirected,22,1
Rhaegar-Targaryen,Sansa-Stark,Undirected,3,1
Rhaegar-Targaryen,Tywin-Lannister,Undirected,4,1
Rhaegar-Targaryen,Viserys-Targaryen,Undirected,5,1
Rickard-Karstark,Robb-Stark,Undirected,9,1
Rickard-Karstark,Roose-Bolton,Undirected,3,1
Rickon-Stark,Robb-Stark,Undirected,20,1
Rickon-Stark,Sansa-Stark,Undirected,8,1
Rickon-Stark,Tyrion-Lannister,Undirected,4,1
Robb-Stark,Robert-Baratheon,Undirected,6,1
Robb-Stark,Rodrik-Cassel,Undirected,15,1
Robb-Stark,Roose-Bolton,Undirected,6,1
Robb-Stark,Samwell-Tarly,Undirected,4,1
Robb-Stark,Sansa-Stark,Undirected,13,1
Robb-Stark,Stiv,Undirected,3,1
Robb-Stark,Theon-Greyjoy,Undirected,37,1
Robb-Stark,Torrhen-Karstark,Undirected,3,1
Robb-Stark,Tyrion-Lannister,Undirected,11,1
Robb-Stark,Tywin-Lannister,Undirected,9,1
Robb-Stark,Walder-Frey,Undirected,7,1
Robb-Stark,Wendel-Manderly,Undirected,3,1
Robb-Stark,Yoren,Undirected,3,1
Robert-Arryn,Tyrion-Lannister,Undirected,9,1
Robert-Baratheon,Sandor-Clegane,Undirected,9,1
Robert-Baratheon,Sansa-Stark,Undirected,26,1
Robert-Baratheon,Stannis-Baratheon,Undirected,21,1
Robert-Baratheon,Tommen-Baratheon,Undirected,3,1
Robert-Baratheon,Tyrion-Lannister,Undirected,18,1
Robert-Baratheon,Tywin-Lannister,Undirected,13,1
Robert-Baratheon,Varys,Undirected,36,1
Robert-Baratheon,Vayon-Poole,Undirected,3,1
Robert-Baratheon,Viserys-Targaryen,Undirected,7,1
Rodrik-Cassel,Theon-Greyjoy,Undirected,6,1
Rodrik-Cassel,Tyrion-Lannister,Undirected,10,1
Rodrik-Cassel,Willis-Wode,Undirected,7,1
Roose-Bolton,Tywin-Lannister,Undirected,4,1
Sandor-Clegane,Sansa-Stark,Undirected,23,1
Sandor-Clegane,Tyrion-Lannister,Undirected,5,1
Sansa-Stark,Stannis-Baratheon,Undirected,3,1
Sansa-Stark,Tomard,Undirected,3,1
Sansa-Stark,Tyrion-Lannister,Undirected,3,1
Sansa-Stark,Varys,Undirected,4,1
Shae,Tyrion-Lannister,Undirected,12,1
Shagga,Timett,Undirected,4,1
Shagga,Tyrion-Lannister,Undirected,17,1
Shagga,Ulf-son-of-Umar,Undirected,4,1
Stannis-Baratheon,Tywin-Lannister,Undirected,4,1
Stannis-Baratheon,Varys,Undirected,5,1
Stevron-Frey,Walder-Frey,Undirected,3,1
Theon-Greyjoy,Tyrion-Lannister,Undirected,3,1
Theon-Greyjoy,Tywin-Lannister,Undirected,3,1
Theon-Greyjoy,Walder-Frey,Undirected,4,1
Timett,Tyrion-Lannister,Undirected,3,1
Tommen-Baratheon,Tyrion-Lannister,Undirected,3,1
Tyrion-Lannister,Tywin-Lannister,Undirected,40,1
Tyrion-Lannister,Vardis-Egen,Undirected,9,1
Tyrion-Lannister,Varys,Undirected,3,1
Tyrion-Lannister,Willis-Wode,Undirected,4,1
Tyrion-Lannister,Yoren,Undirected,10,1
Tywin-Lannister,Varys,Undirected,4,1
Tywin-Lannister,Walder-Frey,Undirected,8,1
Waymar-Royce,Will-(prologue),Undirected,18,1
"""

Create sample edges_df and nodes_df

from io import StringIO

import pandas as pd

got_characters_book_1_io = StringIO(got_characters_book_1_string)
got_characters_book_1 = pd.read_csv(got_characters_book_1_io)
got_characters_book_1.head()

	Source	Target	Type	weight	book
0	Addam-Marbrand	Jaime-Lannister	Undirected	3	1
1	Addam-Marbrand	Tywin-Lannister	Undirected	6	1
2	Aegon-I-Targaryen	Daenerys-Targaryen	Undirected	5	1
3	Aegon-I-Targaryen	Eddard-Stark	Undirected	4	1
4	Aemon-Targaryen-(Maester-Aemon)	Alliser-Thorne	Undirected	4	1

edges_df = got_characters_book_1.drop(columns=['book', 'Type'])
edges_df.columns = edges_df.columns.str.lower()
edges_df.source = edges_df.source.str.replace('-(Maester-Aemon)', '', regex=False)
edges_df.target = edges_df.target.str.replace('-(Maester-Aemon)', '', regex=False)
edges_df.head(20)

	source	target	weight
0	Addam-Marbrand	Jaime-Lannister	3
1	Addam-Marbrand	Tywin-Lannister	6
2	Aegon-I-Targaryen	Daenerys-Targaryen	5
3	Aegon-I-Targaryen	Eddard-Stark	4
4	Aemon-Targaryen	Alliser-Thorne	4
5	Aemon-Targaryen	Bowen-Marsh	4
6	Aemon-Targaryen	Chett	9
7	Aemon-Targaryen	Clydas	5
8	Aemon-Targaryen	Jeor-Mormont	13
9	Aemon-Targaryen	Jon-Snow	34
10	Aemon-Targaryen	Samwell-Tarly	5
11	Aerys-II-Targaryen	Brandon-Stark	4
12	Aerys-II-Targaryen	Eddard-Stark	10
13	Aerys-II-Targaryen	Gerold-Hightower	3
14	Aerys-II-Targaryen	Jaime-Lannister	5
15	Aerys-II-Targaryen	Jon-Arryn	3
16	Aerys-II-Targaryen	Robert-Baratheon	12
17	Aggo	Daenerys-Targaryen	11
18	Aggo	Drogo	6
19	Aggo	Jhogo	4

edges_df.sort_values('weight', ascending=False)

	source	target	weight
329	Eddard-Stark	Robert-Baratheon	291
134	Bran-Stark	Robb-Stark	112
62	Arya-Stark	Sansa-Stark	104
249	Daenerys-Targaryen	Drogo	101
479	Joffrey-Baratheon	Sansa-Stark	87
...	...	...	...
542	Loras-Tyrell	Mace-Tyrell	3
290	Drogo	Quaro	3
285	Drogo	Jommo	3
546	Loras-Tyrell	Rodrik-Cassel	3
0	Addam-Marbrand	Jaime-Lannister	3

684 rows × 3 columns

edges_df.groupby('target').agg(sum).sort_values('weight', ascending=False)

	weight
target
Robert-Baratheon	805
Tyrion-Lannister	584
Sansa-Stark	532
Jon-Snow	449
Robb-Stark	396
...	...
Moreo-Tumitis	3
Mychel-Redfort	3
Oswell-Whent	3
Paxter-Redwyne	3
Matthar	3

143 rows × 1 columns

all_items = list(set(edges_df.source.tolist() + edges_df.target.tolist()))

node_df = edges_df.groupby('source').agg(sum).reset_index().merge(
    edges_df.groupby('target').agg(sum).reset_index(),
    left_on='source',
    right_on='target',
    how='outer'
)
node_df['name'] = node_df['source'].combine_first(node_df['target'])
node_df['weight'] = node_df['weight_x'].fillna(value=0) + node_df['weight_y'].fillna(value=0)
node_df = node_df[['name', 'weight']].sort_values('weight', ascending=False)

assert node_df.shape[0] == len(all_items)

node_df.head()

	name	weight
42	Eddard-Stark	1284.0
124	Robert-Baratheon	941.0
79	Jon-Snow	784.0
136	Tyrion-Lannister	650.0
128	Sansa-Stark	545.0

import helpsk as hlp
import numpy as np
def create_category(x):
    names = x.split('-')
    if len(names) == 1:
        return 'Other'
    return names[len(names) - 1]
node_df['category'] = node_df['name'].apply(create_category)
node_df['category'] = hlp.pandas.top_n_categories(
    categorical=node_df['category'],
    weights=node_df['weight'],
    weight_function=np.sum,
    top_n=10
)
print(len(node_df['category'].unique()))
node_df['category'].unique()

10

['Stark', 'Baratheon', 'Snow', 'Lannister', 'Targaryen', 'Baelish', 'Other', 'Mormont', 'Arryn', 'Cassel']
Categories (10, object): ['Stark', 'Other', 'Baratheon', 'Lannister', ..., 'Mormont', 'Arryn', 'Baelish', 'Cassel']

---

edges_df.head()

	source	target	weight
0	Addam-Marbrand	Jaime-Lannister	3
1	Addam-Marbrand	Tywin-Lannister	6
2	Aegon-I-Targaryen	Daenerys-Targaryen	5
3	Aegon-I-Targaryen	Eddard-Stark	4
4	Aemon-Targaryen	Alliser-Thorne	4

node_df.head()

	name	weight	category
42	Eddard-Stark	1284.0	Stark
124	Robert-Baratheon	941.0	Baratheon
79	Jon-Snow	784.0	Snow
136	Tyrion-Lannister	650.0	Lannister
128	Sansa-Stark	545.0	Stark

Graph Function

First Implementation

most of the code is adopted from https://github.com/rweng18/midsummer_network/blob/master/midsummer_graph.ipynb

def plot_network_graph(edges: pd.DataFrame,
                       source: str,
                       target: str,
                       directional=False,
                       edges_weight: str=None,
                       nodes: pd.DataFrame=None,
                       node_names: str=None,
                       nodes_weight: str=None,
                       nodes_category: str=None,
                       text_filter: str=None,
                       filter_degree_level: int=0,
                       width=900,
                       height=900,
                       seed=None
                       ):
    """
    Args:
        edges: pd.DataFrame containing a column for the source and target; an optional column for the edge/connection weight
        source: name of the column in the `edges` pd.DataFrame that represents the sources
        target: name of the column in the `edges` pd.DataFrame that represents the targets
        directional: TBD
        edges_weight: name of the column in the `edges` pd.DataFrame that represents the weights
        nodes: pd.DataFrame containing a column for the name of each node; an optional column for the node weight and category
        node_names: name of the column in the `nodes` pd.DataFrame that represents the names
        nodes_weight: name of the column in the `nodes` pd.DataFrame that represents the weights
        nodes_category: name of the column in the `nodes` pd.DataFrame that represents the categories
        text_filter: filter to search the edge source/target columns
        filter_degree_level: controls how much information (i.e. degrees) the filter returns
            0: nodes of first-degree connections (but nodes are only connected to `text_filter`)
            1: in addition to `0`, add edges/connections between first-degree connections
            2: in addition to `1`, add edges and nodes to second-degree connections
        width: the width of the plot
        height: the height of the plot
        seed: seed for random generator
    """
    # most of the code is adopted from https://github.com/rweng18/midsummer_network/blob/master/midsummer_graph.ipynb
    from sklearn.preprocessing import MinMaxScaler
    import plotly.graph_objects as go
    import networkx as nx


    assert nodes is None or node_names is not None

    edges = edges.copy()
    nodes = nodes.copy()

    def make_edge(edge_x, edge_y,
                  #text,
                  line_width, line_color):
        """
        Creates a scatter trace for the edge between x's and y's with given width

        Parameters
        ----------
        edge_x: a tuple of the endpoints' x-coordinates in the form, tuple([x0, x1, None])
        edge_y: a tuple of the endpoints' y-coordinates in the form, tuple([y0, y1, None])
        line_width: the width of the line
        line_color: color of the line

        Returns
        -------
        An edge trace that goes between x0 and x1 with specified width.
        """
        return  go.Scatter(
            x=edge_x,
            y=edge_y,
            line=dict(
                width=line_width,
                color=line_color
            ),
            hoverinfo='text',
            #text=([text]),
            mode='lines'
        )

    if text_filter:

        # get all edges where either the source or the target is equal to the text filter
        first_degree_edges = edges[(edges[source] == text_filter) | (edges[target] == text_filter)]
        # get a list of all the node names associated with edges
        first_degree_node_names = list(set(list(first_degree_edges[source]) + list(first_degree_edges[target])))

        if filter_degree_level == 0:
            # just add first-degree edges and nodes
            edges = first_degree_edges
            nodes = nodes[nodes[node_names].apply(lambda node_name: node_name in first_degree_node_names)]
        elif filter_degree_level == 1:
            # add connections/edges where the source and target are in the list of first_degree_node_names
            # meaning, edges where the source or the target equals the text_filter, or edges where the
            # source and target are in the first_degree_node_names.
            edges = edges[
                (edges[source] == text_filter) |
                (edges[target] == text_filter) |
                (
                    edges[source].apply(lambda value: value in first_degree_node_names) &
                    edges[target].apply(lambda value: value in first_degree_node_names)
                )
            ]
            nodes = nodes[nodes[node_names].apply(lambda node_name: node_name in first_degree_node_names)]
        elif filter_degree_level == 2:
            # here we are adding connections where the source and target are in the list of
            # first_degree_node_names (same as before), but now we are adding edges and nodes
            # where either the source or the target are in the first_degree_node_names (which gets
            # the second degree edges/nodes)
            edges = edges[
                # by definition this includes any edges where the source or target
                # is the text_filter
                # plus everyone else in the first-degree nodes
                edges[source].apply(lambda value: value in first_degree_node_names) |
                edges[target].apply(lambda value: value in first_degree_node_names)
            ]
            second_level_items = list(set(list(edges[source]) + list(edges[target])))
            nodes = nodes[nodes[node_names].apply(lambda node_name: node_name in second_level_items)]
        else:
            raise NotImplementedError()

    if edges_weight:
        edges['edges_size'] = MinMaxScaler(feature_range=(1, 5)).\
            fit_transform(pd.DataFrame(edges.loc[:, edges_weight]))
    else:
        edges['edges_size'] = 1

    if nodes_weight:
        nodes['nodes_size'] = MinMaxScaler(feature_range=(1, 5)).\
            fit_transform(pd.DataFrame(nodes.loc[:, nodes_weight]))
    else:
        nodes['nodes_size'] = 1

    node_color_lookup = None
    if nodes_category:
        color_values = [c.value for c in hlp.color.Colors]
        category_values = nodes[nodes_category].unique().tolist()
        category_values.sort()
        assert len(color_values) > len(category_values)
        node_color_lookup = {category: color for category, color in zip(category_values, color_values[0:len(category_values)])}
        del color_values, category_values

    graph = nx.Graph()
    # Add nodes to graph
    for row_index in range(0, nodes.shape[0]):
        graph.add_node(
            nodes.iloc[row_index][node_names],
            size = nodes.iloc[row_index]['nodes_size'],
            color=node_color_lookup[nodes.iloc[row_index]['category']] if nodes_category else 'gray',
            category=nodes.iloc[row_index]['category'],
        )

    # For each co-appearance between two nodes, add an edge
    for row_index in range(0, edges.shape[0]):
        instance = edges.iloc[row_index]
        graph.add_edge(instance[source],
                       instance[target],
                       weight = instance['edges_size'])

    positions = nx.spring_layout(graph, seed=seed)

    edge_trace = []
    for edge in graph.edges():
        if graph.edges()[edge]['weight'] > 0:
            source_value = edge[0]
            target_value = edge[1]

            x0, y0 = positions[source_value]
            x1, y1 = positions[target_value]
            #text = source_value + '--' + target_value + ': ' + str(graph.edges()[edge]['weight'])

            if text_filter and directional:
                colors = 'red' if source_value == text_filter else 'blue'
            else:
                colors = 'blue'

            trace  = make_edge(
                edge_x=[x0, x1, None],
                edge_y=[y0, y1, None],
                #text=text,
                line_width=0.3 * graph.edges()[edge]['weight'] ** 1.75,
                line_color=colors
            )
            edge_trace.append(trace)

    # Make a node trace
    node_trace = go.Scatter(
        x=[],
        y=[],
        text=[],
        textposition="top center",
        textfont={'size': 10},
        mode='markers+text',
        hoverinfo='none',
        marker=dict(color=[],
                    size=[],
                    line=None)
    )
    # For each node in midsummer, get the position and size and add to the node_trace
    for node in graph.nodes():
        x, y = positions[node]
        node_trace['x'] += tuple([x])
        node_trace['y'] += tuple([y])
        #    print(graph.nodes()[node]['color'])
        node_trace['marker']['color'] += tuple([graph.nodes()[node]['color']])
        node_trace['marker']['size'] += tuple([5*graph.nodes()[node]['size']])
        node_text = '<b>' + node + '</b>'
        if nodes_category:
            node_text += '</br></br><sup>' + graph.nodes()[node]['category'] + '</sup>'
        node_trace['text'] += tuple([node_text])

    layout = go.Layout(
        paper_bgcolor='rgba(0,0,0,0)',
        plot_bgcolor='rgba(0,0,0,0)'
    )
    fig = go.Figure(layout = layout)

    for trace in edge_trace:
        fig.add_trace(trace)

    fig.add_trace(node_trace)
    fig.update_layout(
        showlegend = False,
        width=width,
        height=height
    )
    fig.update_xaxes(showticklabels = False)
    fig.update_yaxes(showticklabels = False)
    return fig

edges_df.query("source == 'Petyr-Baelish' | target == 'Petyr-Baelish'").sort_values('weight', ascending=False).head()

	source	target	weight
320	Eddard-Stark	Petyr-Baelish	81
594	Petyr-Baelish	Robert-Baratheon	30
600	Petyr-Baelish	Varys	29
190	Catelyn-Stark	Petyr-Baelish	22
145	Brandon-Stark	Petyr-Baelish	14

plot_network_graph(
    edges=edges_df,
    source='source',
    target='target',
    directional=False,
    edges_weight='weight',
    nodes=node_df,
    node_names='name',
    nodes_weight='weight',
    nodes_category='category',
    text_filter='Petyr-Baelish',
    filter_degree_level=0,
    width=1000,
    height=1000,
    seed=42,
)

plot_network_graph(
    edges=edges_df,
    source='source',
    target='target',
    directional=False,
    edges_weight='weight',
    nodes=node_df,
    node_names='name',
    nodes_weight='weight',
    nodes_category='category',
    text_filter='Petyr-Baelish',
    filter_degree_level=1,
    width=1000,
    height=1000,
    seed=42,
)

plot_network_graph(
    edges=edges_df,
    source='source',
    target='target',
    directional=False,
    edges_weight='weight',
    nodes=node_df,
    node_names='name',
    nodes_weight='weight',
    nodes_category='category',
    text_filter='Petyr-Baelish',
    filter_degree_level=2,
    width=1000,
    height=1000,
    seed=42,
)

plot_network_graph(
    edges=edges_df,
    source='source',
    target='target',
    directional=False,
    edges_weight='weight',
    nodes=node_df,
    node_names='name',
    nodes_weight='weight',
    nodes_category='category',
    width=1000,
    height=1000,
    seed=42,
)

---

Second Implementation

def plot_network_graph(edges: pd.DataFrame,
                       source: str,
                       target: str,
                       # directional=False,
                       edges_weight: str=None,
                       nodes: pd.DataFrame=None,
                       node_names: str=None,
                       nodes_weight: str=None,
                       nodes_category: str=None,
                       # width=900,
                       # height=900,
                       seed=None
                       ):
    import math

    def add_edge(edge_start, edge_end, edge_x_list, edge_y_list, length_frac=1,
                 arrow_pos = None, arrow_length = 0.025, arrow_angle = 30, dot_size = 20):
        # edge_start and edge_end are lists defining start and end points
        # Edge x and y are lists used to construct the graph
        # arrowAngle and arrowLength define properties of the arrowhead
        # arrowPos is None, 'middle' or 'end' based on where on the edge you want the arrow to appear
        # arrowLength is the length of the arrowhead
        # arrowAngle is the angle in degrees that the arrowhead makes with the edge
        # dotSize is the plotly scatter dot size you are using (used to even out line spacing when you have a mix of edge lengths)

        # Get start and end cartesian coordinates
        x0, y0 = edge_start
        x1, y1 = edge_end

        # Incorporate the fraction of this segment covered by a dot into total reduction
        length = math.sqrt( (x1-x0)**2 + (y1-y0)**2 )
        dot_size_conversion = .0565/20 # length units per dot size
        converted_dot_diameter = dot_size * dot_size_conversion
        length_frac_reduction = converted_dot_diameter / length
        length_frac = length_frac - length_frac_reduction

        # If the line segment should not cover the entire distance, get actual start and end coords
        skip_x = (x1-x0)*(1-length_frac)
        skip_y = (y1-y0)*(1-length_frac)
        x0 = x0 + skip_x/2
        x1 = x1 - skip_x/2
        y0 = y0 + skip_y/2
        y1 = y1 - skip_y/2

        # Append line corresponding to the edge
        edge_x_list.append(x0)
        edge_x_list.append(x1)
        edge_x_list.append(None) # Prevents a line being drawn from end of this edge to start of next edge
        edge_y_list.append(y0)
        edge_y_list.append(y1)
        edge_y_list.append(None)

        # Draw arrow
        if arrow_pos:
            # Find the point of the arrow; assume is at end unless told middle
            pointx = x1
            pointy = y1

            eta = math.degrees(math.atan((x1-x0)/(y1-y0))) if y1!=y0 else 90.0

            if arrow_pos == 'middle' or arrow_pos == 'mid':
                pointx = x0 + (x1-x0)/2
                pointy = y0 + (y1-y0)/2

            # Find the directions the arrows are pointing
            signx = (x1-x0)/abs(x1-x0) if x1!=x0 else +1    #verify this once
            signy = (y1-y0)/abs(y1-y0) if y1!=y0 else +1    #verified

            # Append first arrowhead
            dx = arrow_length * math.sin(math.radians(eta + arrow_angle))
            dy = arrow_length * math.cos(math.radians(eta + arrow_angle))
            edge_x_list.append(pointx)
            edge_x_list.append(pointx - signx ** 2 * signy * dx)
            edge_x_list.append(None)
            edge_y_list.append(pointy)
            edge_y_list.append(pointy - signx ** 2 * signy * dy)
            edge_y_list.append(None)

            # And second arrowhead
            dx = arrow_length * math.sin(math.radians(eta - arrow_angle))
            dy = arrow_length * math.cos(math.radians(eta - arrow_angle))
            edge_x_list.append(pointx)
            edge_x_list.append(pointx - signx ** 2 * signy * dx)
            edge_x_list.append(None)
            edge_y_list.append(pointy)
            edge_y_list.append(pointy - signx ** 2 * signy * dy)
            edge_y_list.append(None)


        return edge_x_list, edge_y_list

    from sklearn.preprocessing import MinMaxScaler
    import plotly.graph_objects as go
    import networkx as nx

    if edges_weight:
        edges['edges_size'] = MinMaxScaler(feature_range=(1, 3)).\
            fit_transform(pd.DataFrame(edges.loc[:, edges_weight]))
    else:
        edges['edges_size'] = 1

    if nodes_weight:
        nodes['nodes_size'] = MinMaxScaler(feature_range=(1, 5)).\
            fit_transform(pd.DataFrame(nodes.loc[:, nodes_weight]))
    else:
        nodes['nodes_size'] = 1

    node_color_lookup = None
    if nodes_category:
        color_values = [c.value for c in hlp.color.Colors]
        category_values = nodes[nodes_category].unique().tolist()
        category_values.sort()
        assert len(color_values) > len(category_values)
        node_color_lookup = {category: color for category, color in zip(category_values, color_values[0:len(category_values)])}
        del color_values, category_values

    graph = nx.Graph()
    # Add nodes to graph
    for row_index in range(0, nodes.shape[0]):
        graph.add_node(
            nodes.iloc[row_index][node_names],
            size = nodes.iloc[row_index]['nodes_size'],
            color=node_color_lookup[nodes.iloc[row_index]['category']] if nodes_category else 'gray',
            category=nodes.iloc[row_index]['category'],
        )

    # For each co-appearance between two nodes, add an edge
    for row_index in range(0, edges.shape[0]):
        instance = edges.iloc[row_index]
        graph.add_edge(instance[source],
                       instance[target],
                       weight = instance['edges_size'])

    positions = nx.spring_layout(graph, seed=seed)

    # Controls for how the graph is drawn
    node_color = 'black'
    line_width = 1

    node_x = []
    node_y = []
    node_size = [x*10 for x in nodes.nodes_size]

    for node in graph.nodes():
        x, y = positions[node]
        node_x.append(x)
        node_y.append(y)

    # Make a list of edges for plotly, including line segments that result in arrowheads
    #edge_color = ['red' if x == word_filter else 'blue' for x in filtered_network_data['source']]
    edge_traces = []
    for edge in graph.edges():
        edge_x = []
        edge_y = []
        # addEdge(start, end, edge_x, edge_y, lengthFrac=1, arrowPos = None, arrowLength=0.025, arrowAngle = 30, dotSize=20)
        char_1 = edge[0]
        char_2 = edge[1]
        start = list(positions[char_1])
        end = list(positions[char_2])
        edge_x, edge_y = add_edge(start, end, edge_x, edge_y, 1, 'end', .02, 30, 1)


        word_filter = 'update'
        edge_traces += [go.Scatter(x=edge_x, y=edge_y,
                                   opacity=0.3,
                                   line=dict(width=line_width, color='red' if char_1 == word_filter else 'blue'),
                                   hoverinfo='none', mode='lines')]
    node_trace = go.Scatter(
        x=node_x,
        y=node_y,
        mode='markers',
        #text=nodes['app'].values,
        #textposition='top center',
        #textfont=dict(size=10, color='black'),
        opacity=0.5,
        hoverinfo='text',
        marker=dict(showscale=False, color = node_color, size=node_size)#size=nodeSize)
    )
    text_trace = go.Scatter(
        x=node_x,
        y=node_y,
        mode='text',
        text=[f'{x}<br><sup>{y}</sup>' for x, y in zip(nodes[node_names], nodes[nodes_category])],
        textposition='top center',
        textfont=dict(size=11, color='black'),
        #opacity=0.6,
        #    hoverinfo='text',
        #    marker=dict(showscale=False, color = nodeColor, size=nodes_size)#size=nodeSize)
    )

    fig = go.Figure(data=edge_traces + [node_trace, text_trace],
                    layout=go.Layout(
                        showlegend=False,
                        hovermode='closest',
                        margin=dict(b=20,l=5,r=5,t=40),
                        xaxis=dict(showgrid=False, zeroline=False, showticklabels=False),
                        yaxis=dict(showgrid=False, zeroline=False, showticklabels=False))
                    )

    # Note: if you don't use fixed ratio axes, the arrows won't be symmetrical
    fig.update_layout(yaxis = dict(scaleanchor = "x", scaleratio = 1),
                      plot_bgcolor='rgb(255,255,255)',
                      width=900,
                      height=900)

    return fig

plot_network_graph(
    edges=edges_df,
    source='source',
    target='target',
    # directional=False,
    edges_weight='weight',
    nodes=node_df,
    node_names='name',
    nodes_weight='weight',
    nodes_category='category',
    # width=1000,
    # height=1000,
    seed=42,
)